import org.junit.Test

//给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
//
// 
//
// 示例 1： 
//
// 
//
// 
//输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
//输出：6
//解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 
// 
//
// 示例 2： 
//
// 
//输入：height = [4,2,0,3,2,5]
//输出：9
// 
//
// 
//
// 提示： 
//
// 
// n == height.length 
// 0 <= n <= 3 * 104 
// 0 <= height[i] <= 105 
// 
// Related Topics 栈 数组 双指针 动态规划 单调栈 
// 👍 2463 👎 0


//leetcode submit region begin(Prohibit modification and deletion)
class SolutionTrap {
    @Test
    fun main() {
        // println(trap(intArrayOf(0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1)))
        // println(trap(intArrayOf(4,2,0,3,2,5)))
        // println(trap(intArrayOf(4,2,3)))
        // println(trap(intArrayOf(0,1,0,2,1,0,1,3,2,1,2,1)))
        println(trap(intArrayOf(0, 2, 0)))
        println(trap(intArrayOf(2, 0, 2)))
    }

    /***
     * 滑动窗口法
     * 先左至右 再 右至左
     */
    fun trap(height: IntArray): Int {
        var left = 0
        var water = 0
        val dp = BooleanArray(height.size)
        // 找到第一个非0的
        while (left < height.size) {
            if (height[left] > 0) {
                break
            }
            left++
        }
        var right = left + 1
        // 左至右
        while (left < height.size && right < height.size) {
            if (height[right] >= height[left]) {
                // 两个相邻
                if (right - left == 1) {
                    left = right
                    right += 1
                    continue
                }
                val min = height[left].coerceAtMost(height[right])
                left++
                while (left < right) {
                    if (!dp[left]) {
                        water += min - height[left]
                        dp[left] = true
                    }
                    left++
                }
                // right++
                // continue
            }
            right++
        }
        // 右至左
        right = height.size - 1
        while (right > 0) {
            if (height[right] > 0) {
                break
            }
            right--
        }
        left = right - 1

        while (left >= 0 && right >= 0) {
            if (height[left] >= height[right]) {
                // 两个相邻
                if (right - left == 1) {
                    right = left
                    left--
                    continue
                }
                val min = height[left].coerceAtMost(height[right])
                right--
                while (left < right) {
                    if (!dp[right]) {
                        water += min - height[right]
                        dp[right] = true
                    }
                    right--
                }
                // right++
                // continue
            }
            left--
        }
        return water
        TODO()
    }
}
//leetcode submit region end(Prohibit modification and deletion)
